ubuntu 安裝 Nvidia Cuda / Cudnn 相關筆記

tags: 研究室碩一筆記

安裝 Nvidia 驅動

如果需要安裝驅動，先 Nvidia-smi 確認原本 Nvidia 相關驅動套件版本，

首先 sudo lshw -numeric -C display ，查看 GPU 資訊並查詢支援的 CUDA 版本。

1. 清除既有 Nvidia Driver
sudo apt-get purge nvidia*

2. 加入 GPU ppa
sudo add-apt-repository ppa:graphics-drivers

3. Packages 更新
sudo apt-get update
sudo apt upgrade

4. 列出支援的 GPU Driver 版本 (有想安裝的就跳過，直接指定版本)
ubuntu-drivers list

5. 安裝指定的 Nvidia Driver 版本
sudo apt install nvidia-driver-515

6. 重啟
sudo reboot

最後再 Nvidia-smi 檢查有沒有。

安裝 Cuda / cuDNN

ubuntu-12900
系統版本：Ubuntu 20.04.4 LTS
Kernel 版本：GNU/Linux 5.4.0-148-generic x86_64
GPU : 3080ti * 1

以新機器 ubuntu-12900 為例子，輸入指令 Nvidia-smi 後得到，

確認 Nvidia driver version 後，進入 Nvidia 官方網站，

Nvidia 官方網站連結
https://developer.nvidia.com/cuda-toolkit-archive

下載要的版本，例如 CUDA 11.7.0，點進去對應主機資訊後選擇 runfile 檔案進行下載，

下方會出現這些資訊，

照這些指令進行安裝後會出現，

Step1.

Step2.

由於 Nvidia dirver 我們在前面有安裝過了，這邊取消勾選，

Step3.

按下 Install 便會開始安裝，安裝完成後會出現這個資料夾及這些，

然後要開始配置系統文件，

sudo vim ~/.bashrc

在文件的最後一行加上這兩行，路徑中的 cuda-11.7 / nsight-compute-2022.2.0 ，會依照下載版本不同有所改變。

export PATH=/usr/local/cuda-11.7/bin:/usr/local/cuda-11.7/nsight-compute-2022.2.0${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-11.7/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

bashrc 內容

確認修改完後，輸入指令讓配置生效，

source ~/.bashrc

檢查一下 Cuda 是否有成功被獲得，

nvcc -V

然後下載 cudnn ，

cudnn 網址
https://developer.nvidia.com/rdp/cudnn-download

下載符合自己機器的 cuDNN 版本，如 cuDNN 8.9.0 Linux x86_64 (Tar) 解壓縮，

tar -xvf cudnn-linux-x86_64-8.9.0.131_cuda11-archive.tar.xz
cd cudnn-linux-x86_64-8.9.0.131_cuda11-archive

sudo cp include/* /usr/local/cuda-11.7/include
sudo cp lib/libcudnn* /usr/local/cuda-11.7/lib64

sudo chmod a+r /usr/local/cuda-11.7/include/cudnn*
sudo chmod a+r /usr/local/cuda-11.7/lib64/libcudnn*

測試一下，可以得到 cuDNN_version 的計算結果為 8900，
這代表我們安裝的 cuDNN 版本是 8.9.0 沒錯。

cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2